package com.perfma.xtest.tool.leetcode;

import java.util.*;

/**
 * 存在重复元素 II
 * <p>
 * 给你一个整数数组 nums 和一个整数 k ，判断数组中是否存在两个 不同的索引 i 和 j ，满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在，返回 true ；否则，返回 false 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/contains-duplicate-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author: junhui.si
 * @date: 2022-01-19 11:47
 **/
public class Code219 {

    public boolean containsNearbyDuplicate(int[] nums, int k) {
        if (nums.length == 1) {
            return false;
        }

        Map<Integer, List<Integer>> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int value = nums[i];
            List<Integer> list;
            if (map.containsKey(value)) {
                list = map.get(value);
            } else {
                list = new ArrayList<>();
                map.put(value, list);
            }
            list.add(i);
        }

        for (int key : map.keySet()) {
            List<Integer> v = map.get(key);
            if (v.size() <= 1) {
                continue;
            }
            for (int i = 0; i < v.size() - 1; i++) {
                if (v.get(i + 1) - v.get(i) <= k) {
                    return true;
                }
            }
        }
        return false;
    }

    // 滑动窗口
    public boolean containsNearbyDuplicate2(int[] nums, int k) {
        Set<Integer> set = new HashSet<>();

        for (int i = 0; i < nums.length; i++) {
            if (i - k > 0) {
                set.remove(nums[i - k - 1]);
            }
            if (set.add(nums[i])) {
                return true;
            }
        }
        return false;
    }
}
